package com.example.consumer;

import com.alibaba.fastjson.JSON;
import com.example.entity.Order;
import com.rabbitmq.client.AMQP;
import com.rabbitmq.client.Channel;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class OrderSequenceConsumer {

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    // 使用自定义的容器工厂，为 order.sequence.queue 队列设置特定的并发和预取配置
    @RabbitListener(queues = "order.sequence.queue" , containerFactory = "orderSequenceQueueContainerFactory", ackMode = "MANUAL")
    public void handleOrder( Channel channel, Message message) throws IOException {
        // 正确地从 Message 对象中获取 deliveryTag
        long deliveryTag = message.getMessageProperties().getDeliveryTag();
        String correlationId = message.getMessageProperties().getCorrelationId();
        String messageText = new String(message.getBody(), StandardCharsets.UTF_8);
        //处理任务
        Order order = JSON.parseObject(messageText, Order.class);
        try {
            // 幂等性校验
            if (correlationId != null && isProcessed(correlationId)) {
                log.info("消息已处理过，correlationId：{}，直接确认", correlationId);
                channel.basicAck(deliveryTag, false);
                return;
            }

            // 业务处理
            log.info("order.sequence.queue处理订单order：{}，correlationId：{}", order.getUserId(), correlationId);
            processOrder(order);
            
            // 记录已处理的消息ID，防止重复消费（3天之内的）
            if (correlationId != null) {
                markAsProcessed(correlationId, 3);
            }
            
            // 手动确认消息（deliveryTag：消息唯一标识；multiple：是否批量确认）
            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
        } catch (Exception e) {
            log.error("处理订单失败: {}", order, e);

            // 获取重试次数
            Map<String, Object> headers = message.getMessageProperties().getHeaders();
            Integer retryCount = (Integer) headers.getOrDefault("x-retry-count", 0);
            log.info("当前重试次数：{}", retryCount);
            try {
                // 消息处理失败，根据业务需求决定是否重新入队
                if (retryCount < 8) { // 最大重试3次
                    // 重新入队逻辑
                    AMQP.BasicProperties.Builder builder = new AMQP.BasicProperties.Builder();
                    Map<String, Object> newHeaders = new HashMap<>(headers);
                    newHeaders.put("x-retry-count", retryCount + 1);
                    builder.headers(newHeaders);

                    // 重新发布消息
                    channel.basicPublish("order.sequence.exchange", "order.sequence", builder.build(), message.getBody());
                    // 确认原消息
                    channel.basicAck(deliveryTag, false);
                } else {
                    // 达到最大重试次数，拒绝消息并进入死信队列
                    channel.basicNack(deliveryTag, false, false);
                    log.warn("Message rejected after {} retries: {}", retryCount, order);
                }
            } catch (IOException ioException) {
                log.error("处理消息重试失败", ioException);
                try {
                    // 最后的兜底处理
                    channel.basicNack(deliveryTag, false, false);
                } catch (IOException ex) {
                    log.error("最终确认消息失败", ex);
                }
            }
        }
    }

    /**
     * 检查消息是否已处理
     * @param correlationId 消息关联ID
     * @return 是否已处理
     */
    private boolean isProcessed(String correlationId) {
        String key = "order:processed:" + correlationId;
        // 使用 get 方法替代 hasKey，避免阻塞和全量查询
        String value = redisTemplate.opsForValue().get(key);
        return value != null;
    }

    /**
     * 标记消息已处理
     * @param correlationId 消息关联ID
     * @param expireDays 过期时间（天）
     */
    private void markAsProcessed(String correlationId, int expireDays) {
        String key = "order:processed:" + correlationId;
        redisTemplate.opsForValue().set(key, "1", expireDays, TimeUnit.DAYS);
    }

    /**
     * 处理订单业务逻辑
     * @param order 订单对象
     */
    private void processOrder(Order order) {
        // 实际订单处理逻辑，移除了模拟的Thread.sleep
//        log.info("开始处理订单，订单号：{}", order.getId());

        // 这里添加实际的业务处理逻辑
        // 例如：更新数据库、调用其他服务、发送通知等
        // 原来的Thread.sleep(1000)已移除，提高处理效率

        // 模拟实际的业务处理（这里应该替换为真实的业务逻辑）
        // 例如：
        // orderService.updateOrderStatus(order.getId(), OrderStatus.PROCESSING);
        // inventoryService.reduceInventory(order.getProductId(), order.getQuantity());
        
        // 模拟偶尔的处理失败（生产环境中应该根据实际业务逻辑判断）
        if (order.getUserId() != null && order.getUserId().contains("ERROR")) {
            throw new RuntimeException("模拟订单处理失败");
        }

//        log.info("订单处理完成，订单号：{}", order.getUserId());
    }

}